"""
修复face_database.py文件的编码和缩进问题
"""

import os

def fix_file():
    file_path = "src/modules/face_database.py"
    
    # 读取文件内容（使用正确的编码）
    try:
        with open(file_path, 'r', encoding='utf-8') as f:
            content = f.read()
    except UnicodeDecodeError:
        # 如果UTF-8失败，尝试使用其他编码
        with open(file_path, 'r', encoding='gbk') as f:
            content = f.read()
    
    # 修复缩进问题
    # 查找有问题的部分并修复
    lines = content.split('\n')
    
    # 查找包含特定注释的行
    for i, line in enumerate(lines):
        if "# 最老的数据结构（只有编码，没有图片路径）" in line:
            # 确保这一行有正确的缩进（16个空格）
            lines[i] = "                # 最老的数据结构（只有编码，没有图片路径）"
            # 检查下一行
            if i + 1 < len(lines):
                if "# 这种情况下我们无法删除文件，因为没有存储路径信息" in lines[i + 1]:
                    lines[i + 1] = "                # 这种情况下我们无法删除文件，因为没有存储路径信息"
            break
    
    # 重新组合内容
    fixed_content = '\n'.join(lines)
    
    # 写回文件（使用UTF-8编码）
    with open(file_path, 'w', encoding='utf-8') as f:
        f.write(fixed_content)
    
    print("文件已修复")

if __name__ == "__main__":
    fix_file()